package com.ruoqing.gateway.config;

import com.alibaba.csp.sentinel.adapter.gateway.sc.SentinelGatewayFilter;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.http.codec.ServerCodecConfigurer;
import org.springframework.web.reactive.result.view.ViewResolver;

import java.util.Collections;
import java.util.List;


/**
	* @author YaoXian
	* @title: GateWayConfiguration
	* @projectName ruoxi-blog
	* @description: 若兮微服务 - 网关服务 - 网关整合Sentinel实现动态限流
	* @date 2020-11-23 18:23
	*/

@Configuration
class GatewayConfiguration {
		
		private final List<ViewResolver> viewResolvers;
		private final ServerCodecConfigurer serverCodecConfigurer;
		
		public GatewayConfiguration(ObjectProvider<List<ViewResolver>> viewResolversProvider,
		                            ServerCodecConfigurer serverCodecConfigurer) {
				this.viewResolvers = viewResolversProvider.getIfAvailable(Collections::emptyList);
				this.serverCodecConfigurer = serverCodecConfigurer;
		}
		
		@Bean
		@Order(Ordered.HIGHEST_PRECEDENCE)
		public JsonSentinelGatewayBlockExceptionHandler sentinelGatewayBlockExceptionHandler() {
				// Register the block exception handler for Spring Cloud Gateway.
				return new JsonSentinelGatewayBlockExceptionHandler(viewResolvers, serverCodecConfigurer);
		}
		
		@Bean
		@Order(Ordered.HIGHEST_PRECEDENCE)
		public GlobalFilter sentinelGatewayFilter() {
				return new SentinelGatewayFilter();
		}
}